77a3189d40bfcac007d99ac8b9f888dbae32ad07,python/src/com/jetbrains/python/refactoring/surround/surrounders/statements/PyWithIfSurrounder.java,PyWithIfSurrounder,surroundStatement,#Project#Editor#PsiElement[]#,25
Before Change
PyIfStatement ifStatement =
PythonLanguage.getInstance().getElementGenerator().createFromText(project, PyIfStatement.class, "if True:\n ");
final PsiElement parent = elements[0].getParent();
ifStatement.getIfPart().addRange(elements[0], elements[elements.length - 1]);
ifStatement = (PyIfStatement) parent.addBefore(ifStatement, elements[0]);
parent.deleteChildRange(elements[0], elements[elements.length - 1]);
After Change
PyIfStatement ifStatement =
PythonLanguage.getInstance().getElementGenerator().createFromText(project, PyIfStatement.class, "if True:\n ");
final PsiElement parent = elements[0].getParent();
final PyStatementList statementList = ifStatement.getIfPart().getStatementList();
assert statementList != null;
statementList.addRange(elements[0], elements[elements.length - 1]);
ifStatement = (PyIfStatement) parent.addBefore(ifStatement, elements[0]);